import os
import re
#name为txt文件名，number为分离电压或者电流值的行数，key_word为要搜索的关键词
def separate_txt(name,number,key_word):
    file=name+'.txt'
    sum_U=0
    f=open(file,encoding='gbk',errors='ignore')
    result_U=[]
    data1=f.readlines()
    f.close()
    for line in data1:
        data2=line.split()
        if data2.__len__()==0:
            continue
        m=re.findall(key_word,data2[0])
        if m:
            #line=line.replace(key_word1+'A:','')
            result_U.append(line)
            sum_U+=1
        if sum_U==number:
            break
    out_path_U=name+'_'+key_word+'.txt'
    os.makedirs('结果',exist_ok=True)
    f_U=open('结果/'+out_path_U,'w')
    f_U.writelines(result_U)
    f_U.close()
    return None
#从separateTxt.py运行程序所在路径获取后缀为.txt的文件，并返回文件名列表，子目录不查找。
def fileName():
    listFile=[]
    file_dir=os.getcwd()
    files=os.listdir(file_dir)
    for file in files:
        if(file.find('.txt')!=-1):
            file=file.replace('.txt','')
            listFile.append(file)
    return listFile

if __name__=="__main__":
    number=1000
    key_word1='U'
    key_word2='I'
    name=fileName()
    for str in name:
        separate_txt(str,number,key_word1)
        separate_txt(str,number,key_word2)